Extract arguments from string with python function